package 链表与双指针.m03.d12.t02;

import 链表与双指针.ListNode;

class Solution {
    public static void main(String[] args) {
        ListNode head = createLinkedList(31);
        ListNode res = reverseKGroup(head, 4);
        printLinkedList(res);
    }

    /**
     * 创建一个包含从1到n的节点的链表
     *
     * @param n 链表中节点的数量
     * @return 链表的头节点
     */
    public static ListNode createLinkedList(int n) {
        if (n <= 0) return null; // 如果n小于等于0，返回空链表

        ListNode head = new ListNode(1); // 创建头节点
        ListNode current = head; // 当前节点指针初始化为头节点

        for (int i = 2; i <= n; i++) { // 从第二个节点开始创建
            current.next = new ListNode(i); // 创建新节点并连接到当前节点
            current = current.next; // 移动当前节点指针到新创建的节点
        }

        return head; // 返回链表的头节点
    }
    public static void printLinkedList(ListNode head) {
        ListNode current = head; // 从头节点开始
        while (current != null) { // 当当前节点不为空时继续循环
            System.out.print(current.val); // 打印当前节点的值

            if (current.next != null) {
                System.out.print(" -> "); // 如果不是最后一个节点，打印箭头分隔符
            }

            current = current.next; // 移动到下一个节点
        }
        System.out.println(); // 换行
    }
    public static ListNode reverseKGroup(ListNode head, int k) {
        ListNode hair = new ListNode(0);
        hair.next = head;
        // hair用于返回链表的头部
        ListNode pre = hair;
        // 处理a~b之间的节点，pre是a之前那个的节点，nex是b之后的那个节点
        while (head != null) {
            // tail是反转的那部分链表的前面那一个节点开始
            ListNode tail = pre;
            // 查看剩余部分长度是否大于等于 k
            for (int i = 0; i < k; ++i) {
                tail = tail.next;
                if (tail == null) {
                    return hair.next;
                }
            }
            ListNode nex = tail.next;
            // 对head到tail的链表进行翻转，第一个节点指向下一批节点的开始
            ListNode[] reverse = myReverse(head, tail);
            // head是反转节点新的头部
            // tail是反转节点新的尾部
            head = reverse[0];
            tail = reverse[1];
            // 把子链表重新接回原链表
            pre.next = head;
            tail.next = nex;
            pre = tail;
            head = tail.next;
        }

        return hair.next;
    }

    public static ListNode[] myReverse(ListNode head, ListNode tail) {
        ListNode prev = tail.next;
        ListNode p = head;
        while (prev != tail) {
            ListNode nex = p.next;
            p.next = prev;
            prev = p;
            p = nex;
        }
        return new ListNode[]{tail, head};
    }
}